# 导入pygame库，pygame是一个开源的Python库，用于制作2D游戏
import random
import time

import pygame
from pygame.sprite import Sprite

# 定义屏幕的宽度和高度
SCREEN_WIDTH = 850
SCREEN_HEIGHT = 500

# 定义背景颜色，这是一个RGB颜色，(0, 0, 0)代表黑色
BG_COLOR = pygame.Color(0, 0, 0)

# 定义文本颜色，这也是一个RGB颜色，(255, 0, 0)代表红色
TEXT_COLOR = pygame.Color(255, 0, 0)


# 定义一个基类
class BaseItem(Sprite):
    def __init__(self, color, width, height):
        # Call the parent class (Sprite) constructor
        pygame.sprite.Sprite.__init__(self)


# 主要类
class MainGame:
    # 定义一个类变量，表示游戏窗口，所有的窗口实例都将共享这个变量
    window = None
    my_tank = None
    # 创建存储敌方坦克的列表
    enemyTankList = []
    # 定义敌方坦克的数量
    enemyTankCount = 5
    # 存储我方子弹列表
    myBulletList = []
    # 存储敌方子弹的列表
    enemyBulletList = []
    # 存储爆炸效果
    explodeList = []
    # 存储墙壁的列表
    wallList = []

    # 初始化方法，在创建MainGame对象时会被调用
    def __init__(self):
        # 这里没有进行任何操作，只是一个空的初始化方法
        pass

        # 开始游戏的方法，会创建一个新的游戏窗口并进入主循环

    def startGame(self):
        # 初始化pygame的显示模块
        pygame.display.init()

        # 创建一个新的窗口，窗口的大小是[SCREEN_WIDTH, SCREEN_HEIGHT]，窗口的标题是'坦克大战1.03'
        MainGame.window = pygame.display.set_mode([SCREEN_WIDTH, SCREEN_HEIGHT])
        # 初始化我方坦克
        self.createMyTank()
        # 初始化敌方坦克，并将敌方坦克添加到列表中
        self.createEnemyTank()
        # 创建初始化墙壁
        self.createWall()
        pygame.display.set_caption('坦克大战1.03')
        # 游戏主循环，这是一个无限循环，游戏将持续运行直到窗口被关闭或者主方法被调用结束游戏
        while True:
            # 使坦克移动的速度慢一点
            time.sleep(0.02)
            # 用背景颜色填充窗口，清空窗口的显示内容
            MainGame.window.fill(BG_COLOR)
            # 处理事件，获取所有等待的事件并做出响应
            self.getEvent()
            # 在窗口上绘制文本，文本的内容是"敌方坦克剩余数量6"，文本的位置是(10,10)
            MainGame.window.blit(self.getTextSurface("敌方坦克剩余数量%d" % len(MainGame.enemyTankList)), (10, 10))
            # 调用坦克显示方法
            # 判断我方坦克是否存活
            if MainGame.my_tank and MainGame.my_tank.live:
                MainGame.my_tank.displayTank()
            else:
                # 删除我方坦克
                del MainGame.my_tank
                MainGame.my_tank = None
            # 循环遍历敌方坦克列表，展示敌方坦克
            self.blitEnemyTank()
            # 循环遍历显示我方坦克的子弹
            self.blitMyBullet()
            # 循环遍历敌方子弹列表，展示敌方子弹
            self.biltEnemyBullet()
            # 循环遍历爆炸效果，展示爆炸
            self.blitExplode()
            # 循环遍历墙壁列表来展示墙壁
            self.blitWall()
            # 如果坦克的开关是开启，才可以移动
            if MainGame.my_tank and MainGame.my_tank.live:
                if not MainGame.my_tank.stop:
                    # 调用移动方法
                    MainGame.my_tank.move()
                    # 检测我方坦克是否与墙壁发生碰撞
                    MainGame.my_tank.hitWall()
                    # 检测我方坦克是否与敌方坦克发生碰撞
                    MainGame.my_tank.myTank_hit_enemyTank()
            # 更新窗口的显示内容，将所有的改变显示在屏幕上
            pygame.display.update()

    # 循环遍历墙壁列表来展示墙壁
    def blitWall(self):
        for wall in MainGame.wallList:
            # 判断墙壁是否存活
            if wall.live:
                # 调用墙壁类的显示方法
                wall.displayWall()
            else:
                # 从墙壁列表移除
                MainGame.wallList.remove(wall)

    # 初始化墙壁
    def createWall(self):
        for i in range(6):
            wall = Wall(i * 130, 220)
            # 将墙壁添加到列表中
            MainGame.wallList.append(wall)

    # 创建我方坦克的方法
    def createMyTank(self):
        MainGame.my_tank = MyTank(350, 300)
        # 创建Music对象
        music = Music('img(坦克)/start.wav')
        # 播放音乐
        music.play()

    # 初始化敌方坦克，并将敌方坦克添加到列表中
    def createEnemyTank(self):
        top = 100
        # 循环生成敌方坦克
        for i in range(MainGame.enemyTankCount):
            left = random.randint(0, 600)
            speed = random.randint(1, 4)
            enemy = EnemyTank(left, top, speed)
            MainGame.enemyTankList.append(enemy)

    # 循环展示爆炸效果
    def blitExplode(self):
        for explode in MainGame.explodeList:
            # 判断是否活着
            if explode.live:
                # 展示
                explode.displayExpolode()
            else:
                # 不是活着状态
                MainGame.explodeList.remove(explode)

    # 循环遍历敌方坦克列表，展示敌方坦克
    def blitEnemyTank(self):
        for enemyTank in MainGame.enemyTankList:
            # 判断敌方坦克是否活着
            if enemyTank.live:
                enemyTank.displayTank()
                enemyTank.randMove()
                # 调用检测是否与墙壁碰撞
                enemyTank.hitWall()
                # 检测敌方坦克是否与我方坦克发生碰撞
                if MainGame.my_tank and MainGame.my_tank.live:
                    enemyTank.enemmyTank_hit_myTank()
                    # 发射子弹
                enemyBullet = enemyTank.shot()
                # 敌方子弹是否是None,如果不为None则添加到敌方子弹列表中
                if enemyBullet:
                    # 将敌方子弹存储到敌方子弹
                    MainGame.enemyBulletList.append(enemyBullet)
            else:
                # 不活着，从敌方坦克列表中移除
                MainGame.enemyTankList.remove(enemyTank)

    # 循环遍历我方子弹存储列表展示我方子弹
    def blitMyBullet(self):
        for myBullet in MainGame.myBulletList:
            # 判断当前子弹是否或者状态，如果是则进行显示及移动，否则在列表中删除
            if myBullet.live:
                myBullet.displayBullet()
                # 调用子弹的移动方法
                myBullet.move()
                # 检测我方坦克是否与敌方坦克发生碰撞
                myBullet.myBullet_hit_enemyTank()
                # 我方子弹是否与墙壁碰撞
                myBullet.hitWall()

            # 否则在列表中删除
            else:
                MainGame.myBulletList.remove(myBullet)

    # 循环遍历敌方子弹列表，展示敌方子弹
    def biltEnemyBullet(self):
        for enemyBullet in MainGame.enemyBulletList:
            # 判断是否存活
            if enemyBullet.live:
                enemyBullet.displayBullet()
                enemyBullet.move()
                # 调用敌方子弹与我方坦克碰撞的方法
                enemyBullet.enemyBullet_hit_myTank()
                # 敌方子弹是否与墙壁碰撞
                enemyBullet.hitWall()

            else:
                MainGame.enemyBulletList.remove(enemyBullet)

    # 结束类
    def endGame(self):
        print('谢谢使用，欢迎再次使用')
        exit()

        # 获取文本绘制表面的方法，用于在窗口上绘制文本

    def getTextSurface(self, text):
        # 初始化pygame的字体模块，如果之前没有初始化过的话
        pygame.font.init()
        # 使用系统字体创建一个新的字体对象，字体大小是18，字体的颜色是红色
        font = pygame.font.SysFont('kaiti', 18)
        # 使用字体对象将文本绘制成一个新的表面，这个表面可以用于在窗口上绘制文本
        textSurface = font.render(text, True, TEXT_COLOR)
        # 返回绘制的文本表面
        return textSurface

        # 处理事件的方法，用于获取并处理所有的游戏事件

    def getEvent(self):
        # 获取所有等待的事件列表
        eventList = pygame.event.get()
        # 对每一个事件进行处理
        for event in eventList:
            # 如果事件的类型是pygame.QUIT，那么调用endGame方法结束游戏
            if event.type == pygame.QUIT:
                self.endGame()
                # 如果事件的类型是pygame.KEYDOWN，那么判断按下的键是哪一个并进行相应的处理
            if event.type == pygame.KEYDOWN:
                # 当坦克不存在或者死亡
                if not MainGame.my_tank:
                    # 判断按下的是Esc键，让坦克重生
                    if event.key == pygame.K_ESCAPE:
                        # 让坦克重生,调用创建坦克的方法
                        self.createMyTank()
                if MainGame.my_tank and MainGame.my_tank.live:
                    if event.key == pygame.K_LEFT:
                        # 切换方向
                        MainGame.my_tank.direction = 'L'
                        # MainGame.my_tank.move()
                        # 修改坦克的开关状态
                        MainGame.my_tank.stop = False
                        print("按下左键，坦克向左移动")
                    elif event.key == pygame.K_RIGHT:
                        # 切换方向
                        MainGame.my_tank.direction = 'R'
                        # MainGame.my_tank.move()
                        # 修改坦克的开关状态
                        MainGame.my_tank.stop = False
                        print("按下右键，坦克向右移动")
                    elif event.key == pygame.K_UP:
                        # 切换方向
                        MainGame.my_tank.direction = 'U'
                        # MainGame.my_tank.move()
                        # 修改坦克的开关状态
                        MainGame.my_tank.stop = False
                        print("按下上键，坦克向上移动")
                    elif event.key == pygame.K_DOWN:
                        # 切换方向
                        MainGame.my_tank.direction = 'D'
                        # MainGame.my_tank.move()
                        # 修改坦克的开关状态
                        MainGame.my_tank.stop = False
                        print("按下下键，坦克向下移动")
                    elif event.key == pygame.K_SPACE:
                        print("发射子弹")
                        # 如果当前我方子弹列表大小 小于等于3的时候才可以创建
                        if len(MainGame.myBulletList) < 3:
                            # 创建我方坦克发射子弹
                            myBullet = Bullet(MainGame.my_tank)
                            MainGame.myBulletList.append(myBullet)
                            # 添加我方坦克发射音效
                            music = Music('img(坦克)/hit.wav')
                            music.play()

            # 松开方向键，坦克停止移动，修改坦克的开关状态
            if event.type == pygame.KEYUP:
                # 按下上下左右键跟发射子弹不冲突
                if event.key == pygame.K_UP or event.key == pygame.K_DOWN or event.key == pygame.K_LEFT or event.key == pygame.K_RIGHT:
                    if MainGame.my_tank and MainGame.my_tank.live:
                        MainGame.my_tank.stop = True


# 坦克类
class Tank(BaseItem):
    # 添加距离左边上边
    def __init__(self, left, top):
        # 保存加载的图片,上下左右
        self.images = {
            'U': pygame.image.load('img(坦克)/p1tankU.gif'),
            'D': pygame.image.load('img(坦克)/p1tankD.gif'),
            'L': pygame.image.load('img(坦克)/p1tankL.gif'),
            'R': pygame.image.load('img(坦克)/p1tankR.gif'),
        }
        # 方向
        self.direction = 'U'
        # 根据当前图片的方向获取图片 surface
        self.image = self.images[self.direction]
        # 根据图片获取区域
        self.rect = self.image.get_rect()
        # 设置区域的left和top
        self.rect.left = left
        self.rect.top = top
        # 速度 决定移速的快慢
        self.speed = 5
        # 坦克移动的开关
        self.stop = True
        # 是否活着
        self.live = True
        # 原来坐标
        self.oldLeft = self.rect.left
        self.oldTop = self.rect.top

    # 移动
    def move(self):
        # 移动后记录原始的坐标
        self.oldLeft = self.rect.left
        self.oldTop = self.rect.top
        # 判断坦克的方向进行移动
        if self.direction == 'L':
            if self.rect.left > 0:
                self.rect.left -= self.speed

        elif self.direction == 'U':
            if self.rect.top > 0:
                self.rect.top -= self.speed

        elif self.direction == 'D':
            if self.rect.top + self.rect.height < SCREEN_HEIGHT:
                self.rect.top += self.speed

        elif self.direction == 'R':
            if self.rect.left + self.rect.width < SCREEN_WIDTH:
                self.rect.left += self.speed

    # 射击
    def shot(self):
        return Bullet(self)

    def stay(self):
        self.rect.left = self.oldLeft
        self.rect.top = self.oldTop

    # 检测坦克是否与墙壁发生碰撞
    def hitWall(self):
        for wall in MainGame.wallList:
            if pygame.sprite.collide_rect(self, wall):
                # 将坐标设置为移动之前坐标
                self.stay()

    # 展示坦克方法
    def displayTank(self):
        # 获取展示的对象
        self.image = self.images[self.direction]
        # 调用blit方法展示
        MainGame.window.blit(self.image, self.rect)
    # 我方坦克


class MyTank(Tank):
    def __init__(self, left, top):
        super(MyTank, self).__init__(left, top)

    # 检测我方坦克与敌方坦克发生碰撞
    def myTank_hit_enemyTank(self):
        # 循环遍历敌方坦克列表
        for enemyTank in MainGame.enemyTankList:
            if pygame.sprite.collide_rect(self, enemyTank):
                self.stay()

    # 敌方坦克


class EnemyTank(Tank):
    def __init__(self, left, top, speed):
        # 调用父类的初始化方法
        super(EnemyTank, self).__init__(left, top)
        # 加载图片
        self.images = {
            'U': pygame.image.load('img(坦克)/enemy1U.gif'),
            'D': pygame.image.load('img(坦克)/enemy1D.gif'),
            'L': pygame.image.load('img(坦克)/enemy1L.gif'),
            'R': pygame.image.load('img(坦克)/enemy1R.gif'),
        }
        # 方向,随机生成敌方坦克方向
        self.direction = self.randDirection()
        # 根据方向随机获取图片
        self.image = self.images[self.direction]
        # 区域
        self.rect = self.image.get_rect()
        # 对left和top进行赋值
        self.rect.left = left
        self.rect.top = top
        # 速度
        self.speed = speed
        # 移动开关
        self.flag = True
        # 新增一个步数变量step
        self.step = 20

        # 敌方坦克与我方坦克是否发生碰撞

    def enemmyTank_hit_myTank(self):
        if pygame.sprite.collide_rect(self, MainGame.my_tank):
            self.stay()

    # 随机生成敌方坦克方向
    def randDirection(self):
        num = random.randint(1, 4)
        if num == 1:
            return 'U'
        elif num == 2:
            return 'D'
        elif num == 3:
            return 'L'
        elif num == 4:
            return 'R'

    # 敌方坦克随机移动的方式
    def randMove(self):
        if self.step <= 0:
            # 修改方向
            self.direction = self.randDirection()
            # 让步数复位
            self.step = 60
        else:
            self.move()
            # 让步数递减
            self.step -= 1

    # 重写shot()
    def shot(self):
        # 随机生成100以内的数
        num = random.randint(1, 100)
        if num < 10:
            return Bullet(self)


# 子弹类
class Bullet(BaseItem):
    def __init__(self, tank):
        # 加载图片
        self.image = pygame.image.load('img(坦克)/enemymissile.gif')
        # 坦克的方向决定子弹的方向
        self.direction = tank.direction
        # 获取区域
        self.rect = self.image.get_rect()
        # 子弹的left和top与方向有关
        # 设置子弹的位置
        if self.direction == "L":
            # 子弹的位置 = 坦克的位置 - 子弹的宽度
            self.rect.left = tank.rect.left - self.rect.width
            # 子弹的位置 = 坦克的位置 + 坦克的高度/2 - 子弹的高度/2
            self.rect.top = tank.rect.top + tank.rect.height / 2 - self.rect.height / 2
        elif self.direction == "R":
            # 子弹的位置 = 坦克的位置 + 坦克的宽度
            self.rect.left = tank.rect.left + tank.rect.width
            # 子弹的位置 = 坦克的位置 + 坦克的高度/2 - 子弹的高度/2
            self.rect.top = tank.rect.top + tank.rect.height / 2 - self.rect.height / 2
        elif self.direction == "U":
            # 子弹的位置 = 坦克的位置 + 坦克的宽度/2 - 子弹的宽度/2
            self.rect.left = tank.rect.left + tank.rect.width / 2 - self.rect.width / 2
            # 子弹的位置 = 坦克的位置 - 子弹的高度
            self.rect.top = tank.rect.top - self.rect.height
        elif self.direction == "D":
            # 子弹的位置 = 坦克的位置 + 坦克的宽度/2 - 子弹的宽度/2
            self.rect.left = tank.rect.left + tank.rect.width / 2 - self.rect.width / 2
            # 子弹的位置 = 坦克的位置 + 坦克的高度
            self.rect.top = tank.rect.top + tank.rect.height
        # 设置子弹的速度
        self.speed = 6
        # 子弹的状态是否碰到墙壁，如果喷到墙壁，修改此状态
        self.live = True
        # 移动

    def move(self):
        # 判断坦克的方向进行移动
        if self.direction == 'L':
            if self.rect.left > 0:
                self.rect.left -= self.speed
            else:
                # 修改子弹的状态
                self.live = False
        elif self.direction == 'U':
            if self.rect.top > 0:
                self.rect.top -= self.speed
            else:
                # 修改子弹的状态
                self.live = False
        elif self.direction == 'D':
            if self.rect.top + self.rect.height < SCREEN_HEIGHT:
                self.rect.top += self.speed
            else:
                # 修改子弹的状态
                self.live = False
        elif self.direction == 'R':
            if self.rect.left + self.rect.width < SCREEN_WIDTH:
                self.rect.left += self.speed
            else:
                # 修改子弹的状态
                self.live = False

    # 子弹是否碰撞墙壁
    def hitWall(self):
        # 循环遍历墙壁列表
        for wall in MainGame.wallList:
            if pygame.sprite.collide_rect(self, wall):
                # 让子弹消失
                self.live = False
                # 墙壁的生命值减少
                wall.hp -= 1
                if wall.hp <= 0:
                    # 修改墙壁状态
                    wall.live = False

    # 展示子弹方法

    def displayBullet(self):
        # 将图片 surface加载到窗口
        MainGame.window.blit(self.image, self.rect)

    # 我方子弹与敌方坦克的碰撞
    def myBullet_hit_enemyTank(self):
        # 循环遍历敌方坦克列表，判断是否发生碰撞
        for enemyTank in MainGame.enemyTankList:
            if pygame.sprite.collide_rect(enemyTank, self):
                # 修改敌方坦克和我方坦克的状态
                enemyTank.live = False
                self.live = False
                # 创建爆炸对象
                explode = Explode(enemyTank)
                # 将爆炸对象添加到列表中
                MainGame.explodeList.append(explode)

    # 敌方子弹与我方坦克的碰撞
    def enemyBullet_hit_myTank(self):
        if MainGame.my_tank and MainGame.my_tank.live:
            if pygame.sprite.collide_rect(MainGame.my_tank, self):
                # 创建爆炸对象
                explode = Explode(MainGame.my_tank)
                # 将爆炸对象添加到列表中
                MainGame.explodeList.append(explode)
                # 修改敌方坦克和我方坦克的状态
                self.live = False
                MainGame.my_tank.live = False


# 墙壁类
class Wall():
    def __init__(self, left, top):
        # 加载墙壁图片
        self.image = pygame.image.load('img(坦克)/steels.gif')
        # 获取墙壁的区域
        self.rect = self.image.get_rect()
        # 设置位置left,top
        self.rect.left = left
        self.rect.top = top
        # 是否存活
        self.live = True
        # 设置生命值
        self.hp = 3

    # 展示墙壁的方法
    def displayWall(self):
        MainGame.window.blit(self.image, self.rect)

    # 爆炸效果类


class Explode():
    def __init__(self, tank):
        # 爆炸的位置由当前子弹打中坦克位置决定
        self.rect = tank.rect
        self.images = [
            pygame.image.load('img(坦克)/blast0.gif'),
            pygame.image.load('img(坦克)/blast1.gif'),
            pygame.image.load('img(坦克)/blast2.gif'),
            pygame.image.load('img(坦克)/blast3.gif'),
            pygame.image.load('img(坦克)/blast4.gif'),
        ]
        self.step = 0
        self.image = self.images[self.step]
        # 是否活着
        self.live = True

    # 展示爆炸效果方法
    def displayExpolode(self):
        if self.step < len(self.images):
            # 根据索引获得爆炸对象
            self.image = self.images[self.step]
            self.step += 1
            # 添加主窗口
            MainGame.window.blit(self.image, self.rect)
        else:
            # 修改活着状态
            self.live = False
            self.setp = 0


# 音乐类
class Music():
    def __init__(self, filename):
        self.filename = filename
        # 初始化音乐混合器
        pygame.mixer.init()
        # 加载音乐
        pygame.mixer.music.load(self.filename)

    # 播放音乐
    def play(self):
        pygame.mixer.music.play()


if __name__ == '__main__':
    MainGame().startGame()
